__author__ = 'mariia.vitiuk'
from pages.page import Page
from selenium.webdriver.support.select import Select

class MovieInfo(Page):

    @property
    def imdb(self):
        return self.driver.find_element_by_name("imdbid")

    @property
    def imdb_err(self):
        return self.driver.find_element_by_class_name("error")

    @property
    def name(self):
        return self.driver.find_element_by_name("name")

    @property
    def name_err(self):
        return self.driver.find_element_by_name("required error")

    @property
    def aka(self):
        return self.driver.find_element_by_name("aka")

    @property
    def year(self):
        return self.driver.find_element_by_name("year")

    '''
    add select DVD and others
    '''

    @property
    def save_button(self):
        return self.driver.find_element_by_name("submit")
